/*
  WString.h - String library for Wiring & Arduino
  ...mostly rewritten by Paul Stoffregen...
  Copyright (c) 2009-10 Hernando Barragan.  All right reserved.
  Copyright 2011, Paul Stoffregen, paul@pjrc.com

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  Lesser General Public License for more details.

  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
*/

#ifndef String_class_h
#define String_class_h
#ifdef __cplusplus

#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include "avr/pgmspace.h"

// When compiling programs with this class, the following gcc parameters
// dramatically increase performance and memory (RAM) efficiency, typically
// with little or no increase in code size.
//     -felide-constructors
//     -std=c++0x

class __FlashStringHelper;
#define F(string_literal) (reinterpret_cast<const __FlashStringHelper *>(PSTR(string_literal)))

// An inherited class for holding the result of a concatenation.  These
// result objects are assumed to be writable by subsequent concatenations.
class StringSumHelper;

// The string class
class String {
        // use a function pointer to allow for "if (s)" without the
        // complications of an operator bool(). for more information, see:
        // http://www.artima.com/cppsource/safebool.html
        typedef void (String::*StringIfHelperType)() const;
        void StringIfHelper() const {}

    public:
        // constructors
        // creates a copy of the initial value.
        // if the initial value is null or invalid, or if memory allocation
        // fails, the string will be marked as invalid (i.e. "if (s)" will
        // be false).
        String(const char* cstr = "");
        String(const String& str);
        String(const __FlashStringHelper* str);
#if __cplusplus >= 201103L || defined(__GXX_EXPERIMENTAL_CXX0X__)
        String(String&& rval);
        String(StringSumHelper&& rval);
#endif
        explicit String(char c);
        explicit String(unsigned char, unsigned char base = 10);
        explicit String(int, unsigned char base = 10);
        explicit String(unsigned int, unsigned char base = 10);
        explicit String(long, unsigned char base = 10);
        explicit String(unsigned long, unsigned char base = 10);
        explicit String(float, unsigned char decimalPlaces = 2);
        explicit String(double, unsigned char decimalPlaces = 2);
        ~String(void);

        // memory management
        // return true on success, false on failure (in which case, the string
        // is left unchanged).  reserve(0), if successful, will validate an
        // invalid string (i.e., "if (s)" will be true afterwards)
        unsigned char reserve(unsigned int size);
        inline unsigned int length(void) const {return len;}

        // creates a copy of the assigned value.  if the value is null or
        // invalid, or if the memory allocation fails, the string will be
        // marked as invalid ("if (s)" will be false).
        String& operator = (const String& rhs);
        String& operator = (const char* cstr);
        String& operator = (const __FlashStringHelper* str);
#if __cplusplus >= 201103L || defined(__GXX_EXPERIMENTAL_CXX0X__)
        String& operator = (String&& rval);
        String& operator = (StringSumHelper&& rval);
#endif

        // concatenate (works w/ built-in types)

        // returns true on success, false on failure (in which case, the string
        // is left unchanged).  if the argument is null or invalid, the
        // concatenation is considered unsucessful.
        unsigned char concat(const String& str);
        unsigned char concat(const char* cstr);
        unsigned char concat(char c);
        unsigned char concat(unsigned char c);
        unsigned char concat(int num);
        unsigned char concat(unsigned int num);
        unsigned char concat(long num);
        unsigned char concat(unsigned long num);
        unsigned char concat(float num);
        unsigned char concat(double num);
        unsigned char concat(const __FlashStringHelper* str);

        // if there's not enough memory for the concatenated value, the string
        // will be left unchanged (but this isn't signalled in any way)
        String& operator += (const String& rhs)    {concat(rhs); return (*this);}
        String& operator += (const char* cstr)     {concat(cstr); return (*this);}
        String& operator += (char c)           {concat(c); return (*this);}
        String& operator += (unsigned char num)        {concat(num); return (*this);}
        String& operator += (int num)          {concat(num); return (*this);}
        String& operator += (unsigned int num)     {concat(num); return (*this);}
        String& operator += (long num)         {concat(num); return (*this);}
        String& operator += (unsigned long num)    {concat(num); return (*this);}
        String& operator += (float num)        {concat(num); return (*this);}
        String& operator += (double num)       {concat(num); return (*this);}
        String& operator += (const __FlashStringHelper* str) {concat(str); return (*this);}

        friend StringSumHelper& operator + (const StringSumHelper& lhs, const String& rhs);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, const char* cstr);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, char c);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, unsigned char num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, int num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, unsigned int num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, long num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, unsigned long num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, float num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, double num);
        friend StringSumHelper& operator + (const StringSumHelper& lhs, const __FlashStringHelper* rhs);

        // comparison (only works w/ Strings and "strings")
        operator StringIfHelperType() const { return buffer ? &String::StringIfHelper : 0; }
        int compareTo(const String& s) const;
        unsigned char equals(const String& s) const;
        unsigned char equals(const char* cstr) const;
        unsigned char operator == (const String& rhs) const {return equals(rhs);}
        unsigned char operator == (const char* cstr) const {return equals(cstr);}
        unsigned char operator != (const String& rhs) const {return !equals(rhs);}
        unsigned char operator != (const char* cstr) const {return !equals(cstr);}
        unsigned char operator < (const String& rhs) const;
        unsigned char operator > (const String& rhs) const;
        unsigned char operator <= (const String& rhs) const;
        unsigned char operator >= (const String& rhs) const;
        unsigned char equalsIgnoreCase(const String& s) const;
        unsigned char startsWith(const String& prefix) const;
        unsigned char startsWith(const String& prefix, unsigned int offset) const;
        unsigned char endsWith(const String& suffix) const;

        // character acccess
        char charAt(unsigned int index) const;
        void setCharAt(unsigned int index, char c);
        char operator [](unsigned int index) const;
        char& operator [](unsigned int index);
        void getBytes(unsigned char* buf, unsigned int bufsize, unsigned int index = 0) const;
        void toCharArray(char* buf, unsigned int bufsize, unsigned int index = 0) const
        { getBytes((unsigned char*)buf, bufsize, index); }
        const char* c_str() const { return buffer; }
        char* begin() { return buffer; }
        char* end() { return buffer + length(); }
        const char* begin() const { return c_str(); }
        const char* end() const { return c_str() + length(); }

        // search
        int indexOf(char ch) const;
        int indexOf(char ch, unsigned int fromIndex) const;
        int indexOf(const String& str) const;
        int indexOf(const String& str, unsigned int fromIndex) const;
        int lastIndexOf(char ch) const;
        int lastIndexOf(char ch, unsigned int fromIndex) const;
        int lastIndexOf(const String& str) const;
        int lastIndexOf(const String& str, unsigned int fromIndex) const;
        String substring(unsigned int beginIndex) const { return substring(beginIndex, len); };
        String substring(unsigned int beginIndex, unsigned int endIndex) const;

        // modification
        void replace(char find, char replace);
        void replace(const String& find, const String& replace);
        void remove(unsigned int index);
        void remove(unsigned int index, unsigned int count);
        void toLowerCase(void);
        void toUpperCase(void);
        void trim(void);

        // parsing/conversion
        long toInt(void) const;
        float toFloat(void) const;
        double toDouble(void) const;

    protected:
        char* buffer;           // the actual char array
        unsigned int capacity;  // the array length minus one (for the '\0')
        unsigned int len;       // the String length (not counting the '\0')
    protected:
        void init(void);
        void invalidate(void);
        unsigned char changeBuffer(unsigned int maxStrLen);
        unsigned char concat(const char* cstr, unsigned int length);

        // copy and move
        String& copy(const char* cstr, unsigned int length);
        String& copy(const __FlashStringHelper* pstr, unsigned int length);
#if __cplusplus >= 201103L || defined(__GXX_EXPERIMENTAL_CXX0X__)
        void move(String& rhs);
#endif
};

class StringSumHelper : public String {
    public:
        StringSumHelper(const String& s) : String(s) {}
        StringSumHelper(const char* p) : String(p) {}
        StringSumHelper(char c) : String(c) {}
        StringSumHelper(unsigned char num) : String(num) {}
        StringSumHelper(int num) : String(num) {}
        StringSumHelper(unsigned int num) : String(num) {}
        StringSumHelper(long num) : String(num) {}
        StringSumHelper(unsigned long num) : String(num) {}
        StringSumHelper(float num) : String(num) {}
        StringSumHelper(double num) : String(num) {}
};

#endif  // __cplusplus
#endif  // String_class_h
